from utils.Dataset import FaceEmbed
from torch.utils.data import DataLoader
import torch.optim as optim
import visdom
import os
import torch
from network.sexual import sexdiscrimination
import torch.nn as nn
import torch.nn.functional as F
import numpy as np
import time

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")